# -*- coding: utf-8 -*-
"""
Created on Fri Apr  3 14:46:23 2020

@author: wjx
"""
class User:
    def __init__(self, name, pwd):
        self.name = name
        self.pwd = pwd


class Account:
    def __init__(self):
        # 用户列表，数据格式：[user对象，user对象，user对象]
        self.user_list = []

    def login(self):
        """
        用户登录，输入用户名和密码然后去self.user_list中校验用户合法性
        :return:
        """
        print('-----登录------')
        name = input('请输入用户名:')
        pwd = input('请输入密码:')
        user = User(name,pwd)
        if user in self.user_list:
            print('登陆成功')
        else:
            print('失败')

    def register(self):
        """
        用户注册，没注册一个用户就创建一个user对象，然后添加到self.user_list中，表示注册成功。
        :return:
        """
        print('-----注册--------')
        while True:
            name = input('请输入用户名:')
            if name.upper() == 'N':
                break
            pwd = input('请输入密码:')
            user = User(name,pwd)
            self.user_list.append(user)
        

    def run(self):
        """
        主程序
        :return:
        """
        choice = input('1.注册\n2.登录')
        if choice == '1':
            self.register()
        else:
            self.login()
        


if __name__ == '__main__':
    obj = Account()
    obj.run()

